Skip to content

[webgpu] Enable profiling for graph capture#27058

Merged
fs-eire merged 10 commits intomainfrom
graph_profiling
Feb 5, 2026
Merged

[webgpu] Enable profiling for graph capture#27058
fs-eire merged 10 commits intomainfrom
graph_profiling

Conversation

@qjia7
Copy link
Contributor

@qjia7 qjia7 commented Jan 19, 2026

Description

Motivation and Context

Copy link
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

This PR enables profiling support during graph capture and replay in the WebGPU execution provider. Previously, profiling was explicitly prevented during graph capture mode.

Changes:

  • Removed the enforcement that blocked profiling during graph capture
  • Added profiling data storage in captured commands for replay
  • Wrapped graph replay with profiling start/collect calls

Reviewed changes

Copilot reviewed 3 out of 3 changed files in this pull request and generated 1 comment.

File Description
onnxruntime/core/providers/webgpu/webgpu_execution_provider.cc Added profiling calls around graph replay to enable profiling data collection
onnxruntime/core/providers/webgpu/webgpu_context.h Added optional profiling data field to CapturedCommandInfo and new PendingKernelInfo constructor for replay
onnxruntime/core/providers/webgpu/webgpu_context.cc Stores profiling data during capture and restores it during replay; removed profiling restriction

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Copy link
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

Copilot reviewed 3 out of 3 changed files in this pull request and generated 2 comments.


💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Copy link
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

Copilot reviewed 3 out of 3 changed files in this pull request and generated 1 comment.


💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

Copy link
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

Copilot reviewed 3 out of 3 changed files in this pull request and generated 3 comments.


💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

@qjia7
Copy link
Contributor Author

qjia7 commented Jan 28, 2026

@guschmue @fs-eire @xiaofeihan1 Please take a look, thanks.

@qjia7 qjia7 requested a review from xiaofeihan1 January 29, 2026 01:52
xiaofeihan1
xiaofeihan1 previously approved these changes Jan 29, 2026
Copy link
Contributor

@xiaofeihan1 xiaofeihan1 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM.

@guschmue guschmue added the ep:WebGPU ort-web webgpu provider label Jan 29, 2026
@fs-eire
Copy link
Contributor

fs-eire commented Feb 5, 2026

please merge the latest main again to include all fixes to the pipelines.

@fs-eire fs-eire enabled auto-merge (squash) February 5, 2026 07:57
@fs-eire fs-eire merged commit b7c183a into main Feb 5, 2026
90 checks passed
@fs-eire fs-eire deleted the graph_profiling branch February 5, 2026 09:21
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ep:WebGPU ort-web webgpu provider

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants